﻿//  ------------------------------------------------------------------------------------
//  Copyright (c) Microsoft Corporation
//  All rights reserved. 
//  
//  Licensed under the Apache License, Version 2.0 (the ""License""); you may not use this 
//  file except in compliance with the License. You may obtain a copy of the License at 
//  http://www.apache.org/licenses/LICENSE-2.0  
//  
//  THIS CODE IS PROVIDED *AS IS* BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, 
//  EITHER EXPRESS OR IMPLIED, INCLUDING WITHOUT LIMITATION ANY IMPLIED WARRANTIES OR 
//  CONDITIONS OF TITLE, FITNESS FOR A PARTICULAR PURPOSE, MERCHANTABLITY OR 
//  NON-INFRINGEMENT. 
// 
//  See the Apache Version 2.0 License for specific language governing permissions and 
//  limitations under the License.
//  ------------------------------------------------------------------------------------

namespace Amqp.Listener
{
    using System.Collections.Concurrent;

    class LinkCollection
    {
        readonly ConcurrentDictionary<LinkId, ListenerLink> links;

        public LinkCollection(string containerId)
        {
            this.links = new ConcurrentDictionary<LinkId, ListenerLink>();
        }

        public bool TryAdd(ListenerLink link)
        {
            var key = LinkId.Create(link.Session.Connection, link.Role, link.Name);
            return this.links.TryAdd(key, link);
        }

        public bool Remove(ListenerLink link)
        {
            var key = LinkId.Create(link.Session.Connection, link.Role, link.Name);
            ListenerLink temp;
            return this.links.TryRemove(key, out temp);
        }

        public void Clear()
        {
            this.links.Clear();
        }
    }
}
